python train.py --n-labeled 1000 --out mnist@1000new --dataset 'mnist'